How To Test And Train Data In Python